package com.atguigu.tingshu.user.service.impl;

import com.alibaba.fastjson.JSON;
import com.atguigu.tingshu.common.constant.KafkaConstant;
import com.atguigu.tingshu.common.constant.SystemConstant;
import com.atguigu.tingshu.common.service.KafkaService;
import com.atguigu.tingshu.common.util.MongoUtil;
import com.atguigu.tingshu.model.user.UserListenProcess;
import com.atguigu.tingshu.user.service.UserListenProcessService;
import com.atguigu.tingshu.vo.album.TrackStatMqVo;
import com.atguigu.tingshu.vo.user.UserListenProcessVo;
import org.bson.types.ObjectId;
import org.springframework.beans.BeanUtils;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.data.domain.Sort;
import org.springframework.data.mongodb.core.MongoTemplate;
import org.springframework.data.mongodb.core.query.Criteria;
import org.springframework.data.mongodb.core.query.Query;
import org.springframework.data.redis.core.RedisTemplate;
import org.springframework.stereotype.Service;

import java.math.BigDecimal;
import java.util.*;
import java.util.concurrent.TimeUnit;

@Service
@SuppressWarnings({"unchecked", "rawtypes"})
public class UserListenProcessServiceImpl implements UserListenProcessService {

    @Autowired
    private MongoTemplate mongoTemplate;

	@Autowired
	private RedisTemplate redisTemplate;

	@Autowired
	private KafkaService kafkaService;


	@Override
	public Map<String, Object> getLatelyTrack(Long userId) {
		//	创建map 集合
		Map<String, Object> map = new HashMap<>();
		//	创建Query 对象
		Query query = Query.query(Criteria.where("userId").is(userId)).with(Sort.by(Sort.Direction.DESC,"updateTime"));
		UserListenProcess userListenProcess = this.mongoTemplate.findOne(query, UserListenProcess.class, MongoUtil.getCollectionName(MongoUtil.MongoCollectionEnum.USER_LISTEN_PROCESS, userId));
		//	判断对象
		if (null != userListenProcess){
			map.put("albumId",userListenProcess.getAlbumId());
			map.put("trackId",userListenProcess.getTrackId());
		}
		//	返回数据
		return map;
	}

	@Override
    public void updateListenProcess(Long userId, UserListenProcessVo userListenProcessVo) {
        //	创建一个存储声音对象
        UserListenProcess userListenProcess = null;
        //	判断MongoDB中是否已经存储声音对象
        Query query = new Query();
        query.addCriteria(Criteria.where("trackId").is(userListenProcessVo.getTrackId()));
        userListenProcess = this.mongoTemplate.findOne(query, UserListenProcess.class, MongoUtil.getCollectionName(MongoUtil.MongoCollectionEnum.USER_LISTEN_PROCESS, userId));
        //	判断
        if (null != userListenProcess) {
			//	更新
			userListenProcess.setBreakSecond(userListenProcessVo.getBreakSecond());
			userListenProcess.setUpdateTime(new Date());
        } else {
			//	新增
			userListenProcess = new UserListenProcess();
			//	属性拷贝
			BeanUtils.copyProperties(userListenProcessVo,userListenProcess);
			//	设置其他属性
			userListenProcess.setId(ObjectId.get().toString());
			//	userListenProcess.setId(UUID.randomUUID().toString().replaceAll("-",""));
			userListenProcess.setUserId(userId);
			userListenProcess.setIsShow(1);
			userListenProcess.setCreateTime(new Date());
			userListenProcess.setUpdateTime(new Date());
		}
		//	保存或修改
        mongoTemplate.save(userListenProcess, MongoUtil.getCollectionName(MongoUtil.MongoCollectionEnum.USER_LISTEN_PROCESS, userId));
		//	防止用户刷声音播放量. 采用userId 与 trackId 统计声音播放次数。
		//	定义一个key setbit key offset value;
		String key = "user:track:" + userId;
		//	获取数据
		Boolean exist = this.redisTemplate.opsForValue().getBit(key, userListenProcessVo.getTrackId());
		//	exist=true; 获取到数据 exist=false; 没有获取到数据
		if (!exist){
			//	记录当前这个用户，第一次播放这个声音
			this.redisTemplate.opsForValue().setBit(key,userListenProcessVo.getTrackId(),true);
			//	给key设置过期时间
			this.redisTemplate.expire(key,24, TimeUnit.HOURS);
			//	修改当前这个声音与专辑对应的播放量.
			TrackStatMqVo trackStatMqVo = new TrackStatMqVo();
			//	赋值：
			//	业务编号：防止消息重复消费。
			trackStatMqVo.setBusinessNo(UUID.randomUUID().toString().replaceAll("-",""));
			trackStatMqVo.setStatType(SystemConstant.TRACK_STAT_PLAY);
			trackStatMqVo.setTrackId(userListenProcessVo.getTrackId());
			trackStatMqVo.setCount(1);
			trackStatMqVo.setAlbumId(userListenProcessVo.getAlbumId());
			//	发送消息
			kafkaService.sendMsg(KafkaConstant.QUEUE_TRACK_STAT_UPDATE, JSON.toJSONString(trackStatMqVo));
		}
	}

    @Override
    public BigDecimal getTrackBreakSecond(Long trackId, Long userId) {
        //	保存时，将数据封装到 UserListenProcess  userId=组成集合名称. userListenProcess_userId
        Query query = Query.query(Criteria.where("trackId").is(trackId));
        //	获取数据
        UserListenProcess userListenProcess = this.mongoTemplate.findOne(query, UserListenProcess.class, MongoUtil.getCollectionName(MongoUtil.MongoCollectionEnum.USER_LISTEN_PROCESS, userId));
        //	判断
        if (null != userListenProcess) {
            return userListenProcess.getBreakSecond();
        }
        //	默认返回
        return new BigDecimal("0");
    }
}
